package com.itany.mall.jdbc.template;

import java.util.List;

/**
 * JDBC模板辅助类, 用于存储返回主键
 * @author 赵云
 *
 */
public class KeyHolder<K> {

	private List<K> keys;

	/**
	 * 存储keys
	 * @param keys 要存储的主键
	 */
	public void setKeys(List<K> keys) {
		this.keys = keys;
	}

	/**
	 * 如果有多个主键, 通过该方法获取出所有主键
	 * @return
	 */
	public List<K> getKeys() {
		return this.keys;
	}

	/**
	 * 如果只有一个主键, 通过该方法获取
	 * @return
	 */
	public K getKey() {
		return this.keys != null && this.keys.size() > 0 ? this.keys.get(0) : null;
	}

}
